#-*- coding: utf-8 -*-
#! /usr/bin/python

import sys
import os
import time
from PAM30 import PAMIE
import urllib

reload(sys)
sys.setdefaultencoding("utf-8")


def savefile(imgurl, folder):
    '''
    存储指定图片url为文件到本地
    默认为jpg，时间命名
    '''
    if os.path.exists(folder)==False:
        os.makedirs(folder)
    fname = time.time()
    ay = imgurl.split('.')
    tail = ay[len(ay)-1]
    try:
        if(urllib.urlretrieve(imgurl, folder+str(fname)+'.'+tail)):
            time.sleep(0.5)
            print 'save '+imgurl+' ok'
    except:
        print 'save error'        


def getimg(url, folder):
    '''
    获取指定网址内容中的所有img的url
    '''
    ie.navigate(url)
    urllist = ie.getImagesValue('src')
    for element in urllist:
        #print element
        savefile(element, folder)


#url = 'http://mooogu.cn/blog/zhipingdmeihao.html'
print "*******************************************************************"
print "在dos下使用本文件，带参数（可以copy网址，dos窗口右键粘贴即可）".encode('gbk')
print "copy 本文件到d盘根目录，进入dos，进入d盘，输入getimg 空格 商品网址".encode('gbk')
print "如 http://item.taobao.com/item.htm?id=6758328991".encode('gbk')
print "**********自动抓取指定网址内的所有图片到D:\\tmp目录下**************".encode('gbk')
print "****************定制各种淘宝营销软件，QQ:91377*********************".encode('gbk')
print "*******************************************************************"
try:
    url = sys.argv[1]
    folder = 'd:\\tmp\\'
    ie= PAMIE()
    getimg(url, folder)
except:
    print "在dos下使用本文件，带参数（可以copy网址，dos窗口右键粘贴即可".encode('gbk')
    print "copy 本文件到d盘根目录，进入dos，进入d盘，输入getimg 空格 商品网址".encode('gbk')
    time.sleep(10)
